package com.itdfq.delay.config;

import com.itdfq.delay.constant.RabbitMqConstant;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

/**
 * @Author: QianMo
 * @Date: 2021/10/21 16:36
 * @Description:
 */
@Configuration
public class RabbitDelayConfig {
    /**
     * 延时队列
     *
     *
     * @return
     */
    @Bean
    public Queue DelayQueue() {
        Map<String, Object> params = new HashMap<>();
        // x-dead-letter-exchange 声明了队列里的死信转发到的DLX名称，
        params.put("x-dead-letter-exchange", RabbitMqConstant.IMMEDIATE_EXCHANGE);
        // x-dead-letter-routing-key 声明了这些死信在转发时携带的 routing-key 名称。
        params.put("x-dead-letter-routing-key", RabbitMqConstant.IMMEDIATE_ROUTING_KEY);
        // x-message-ttl 声明该队列死信可存活时间
        params.put("x-message-ttl", RabbitMqConstant.DELAY_TIME);
        return new Queue(RabbitMqConstant.DELAY_QUEUE, true, false, false, params);
    }


    /**
     * 延时死信Exchange
     *
     * @return
     */
    @Bean
    public DirectExchange DeadLetterExchange() {
        // 一共有三种构造方法，可以只传exchange的名字， 第二种，可以传exchange名字，是否支持持久化，是否可以自动删除，
        // 第三种在第二种参数上可以增加Map，Map中可以存放自定义exchange中的参数
        return new DirectExchange(RabbitMqConstant.DELAY_EXCHANGE, true, false);
    }

    /**
     * 把延时消费的队列和延时消费的exchange绑定在一起
     *
     * @return
     */
    @Bean
    public Binding durotarInner24HoursDelayBinding() {
        return BindingBuilder.bind(DelayQueue()).to(DeadLetterExchange()).with(RabbitMqConstant.DELAY_ROUTING_KEY);
    }


    /****************可变延时消费*******************/

    /**
     * 延时队列，可变延时
     *
     * @return
     */
    @Bean
    public Queue variableDelayQueue() {
        Map<String, Object> params = new HashMap<>();
        // x-dead-letter-exchange 声明了队列里的死信转发到的DLX名称，
        params.put("x-dead-letter-exchange", RabbitMqConstant.IMMEDIATE_EXCHANGE);
        // x-dead-letter-routing-key 声明了这些死信在转发时携带的 routing-key 名称。
        params.put("x-dead-letter-routing-key", RabbitMqConstant.IMMEDIATE_ROUTING_KEY);
        return new Queue(RabbitMqConstant.DELAY_VARIABLE_QUEUE_KEY, true, false, false, params);
    }


    /**
     * 延时 exchange，可变延时
     *
     * @return
     */
    @Bean
    public DirectExchange variableDelayExchange() {
        // 一共有三种构造方法，可以只传exchange的名字， 第二种，可以传exchange名字，是否支持持久化，是否可以自动删除，
        // 第三种在第二种参数上可以增加Map，Map中可以存放自定义exchange中的参数
        return new DirectExchange(RabbitMqConstant.DELAY_VARIABLE_EXCHANGE_KEY, true, false);
    }

    /**
     * 延时 binding，可变延时
     *
     * @return
     */
    @Bean
    public Binding variableDelayBinding() {
        return BindingBuilder.bind(variableDelayQueue()).to(variableDelayExchange()).with(RabbitMqConstant.DELAY_VARIABLE_ROUTING_KEY);
    }
}
